924. Minimize Malware Spread.go

package leetcode

import (
"math"

"github.com/halfrost/LeetCode-Go/template"
)

funcminMalwareSpread(graph [][]int, initial []int) int {
iflen(initial) ==0 {
return0
 }
uf, maxLen, maxIdx, uniqInitials, compMap:= template.UnionFindCount{}, math.MinInt32, -1, map[int]int{}, map[int][]int{}
uf.Init(len(graph))
fori:=rangegraph {
forj:=i+1; j<len(graph); j++ {
ifgraph[i][j] ==1 {
uf.Union(i, j)
 }
 }
 }
for_, i:=rangeinitial {
compMap[uf.Find(i)] =append(compMap[uf.Find(i)], i)
 }
for_, v:=rangecompMap {
iflen(v) ==1 {
uniqInitials[v[0]] =v[0]
 }
 }
iflen(uniqInitials) ==0 {
smallestIdx:=initial[0]
for_, i:=rangeinitial {
ifi<smallestIdx {
smallestIdx=i
 }
 }
returnsmallestIdx
 }
for_, i:=rangeinitial {
if_, ok:=uniqInitials[i]; ok {
size:=uf.Count()[uf.Find(i)]
ifmaxLen<size {
maxLen, maxIdx=size, i
 } elseifmaxLen==size {
ifi<maxIdx {
maxIdx=i
 }
 }
 }
 }
returnmaxIdx
}